package com.licheedev.glide_qrcode;

import android.graphics.Bitmap;
import android.graphics.Canvas;
import android.graphics.Paint;
import android.text.TextUtils;
import androidx.annotation.Nullable;
import com.google.zxing.BarcodeFormat;
import com.google.zxing.EncodeHintType;
import com.google.zxing.WriterException;
import com.google.zxing.common.BitMatrix;
import com.google.zxing.qrcode.QRCodeWriter;
import java.util.Hashtable;

public class QRCodeUtil {

    public static Bitmap loadQrCode(QrCode qrCode) throws Exception {

        Bitmap logo = null;
        Bitmap bitmapBlack = null;
        if (qrCode.isToLogoLogo() && qrCode.getLogo() != null) {
            logo = qrCode.getBitmapFetcher().toBitmap(qrCode.getLogo());
        }

        if (qrCode.getContentReplace() != null) {
            bitmapBlack = qrCode.getBitmapFetcher().toBitmap(qrCode.getContentReplace());
        }

        if (qrCode.isImageUrl()) {
            Bitmap bitmap = qrCode.getBitmapFetcher().toBitmap(qrCode.getQr());
            return addLogo(bitmap, logo, qrCode.getLogoPercent(), qrCode.getLogoBackgroundColor());
        } else {
            return createQRCodeBitmap(qrCode.getQr(), qrCode.getSize(), qrCode.getSize(),
                qrCode.getCharSet(), qrCode.getErrorCorrection(), qrCode.getMarginText(),
                qrCode.getContentColor(), qrCode.getBackgroundColor(), logo,
                qrCode.getLogoPercent(), qrCode.getLogoBackgroundColor(), bitmapBlack);
        }
    }

    /**
     * 生成自定义二维码
     *
     * @param content 字符串内容
     * @param width 二维码宽度
     * @param height 二维码高度
     * @param charset 编码方式（一般使用UTF-8）
     * @param errorCorrectionLevel 容错率 L：7% M：15% Q：25% H：35%
     * @param margin 空白边距（二维码与边框的空白区域）
     * @param colorBlack 黑色色块
     * @param colorWhite 白色色块
     * @param logoBitmap logo图片（传null时不添加logo）
     * @param logoPercent logo所占百分比
     * @param bitmapBlack 用来代替黑色色块的图片（传null时不代替）
     * @return
     */
    public static Bitmap createQRCodeBitmap(String content, int width, int height, String charset,
        String errorCorrectionLevel, String margin, int colorBlack, int colorWhite,
        Bitmap logoBitmap, float logoPercent, int logoBackgroundColor, Bitmap bitmapBlack) {
        // 字符串内容判空
        if (TextUtils.isEmpty(content)) {
            return null;
        }
        // 宽和高>=0
        if (width < 0 || height < 0) {
            return null;
        }
        try {
            /** 1.设置二维码相关配置,生成BitMatrix(位矩阵)对象 */
            Hashtable<EncodeHintType, String> hints = new Hashtable<>();
            // 字符转码格式设置
            if (!TextUtils.isEmpty(charset)) {
                hints.put(EncodeHintType.CHARACTER_SET, charset);
            }
            // 容错率设置
            if (!TextUtils.isEmpty(errorCorrectionLevel)) {
                hints.put(EncodeHintType.ERROR_CORRECTION, errorCorrectionLevel);
            }
            // 空白边距设置
            if (!TextUtils.isEmpty(margin)) {
                hints.put(EncodeHintType.MARGIN, margin);
            }
            /** 2.将配置参数传入到QRCodeWriter的encode方法生成BitMatrix(位矩阵)对象 */
            BitMatrix bitMatrix =
                new QRCodeWriter().encode(content, BarcodeFormat.QR_CODE, width, height, hints);

            /** 3.创建像素数组,并根据BitMatrix(位矩阵)对象为数组元素赋颜色值 */
            if (bitmapBlack != null) {
                //从当前位图按一定的比例创建一个新的位图
                bitmapBlack = Bitmap.createScaledBitmap(bitmapBlack, width, height, false);
            }
            int[] pixels = new int[width * height];
            for (int y = 0; y < height; y++) {
                for (int x = 0; x < width; x++) {
                    //bitMatrix.get(x,y)方法返回true是黑色色块，false是白色色块
                    if (bitMatrix.get(x, y)) {// 黑色色块像素设置
                        if (bitmapBlack != null) {//图片不为null，则将黑色色块换为新位图的像素。
                            pixels[y * width + x] = bitmapBlack.getPixel(x, y);
                        } else {
                            pixels[y * width + x] = colorBlack;
                        }
                    } else {
                        pixels[y * width + x] = colorWhite;// 白色色块像素设置
                    }
                }
            }

            /** 4.创建Bitmap对象,根据像素数组设置Bitmap每个像素点的颜色值,并返回Bitmap对象 */
            Bitmap bitmap = Bitmap.createBitmap(width, height, Bitmap.Config.ARGB_8888);
            bitmap.setPixels(pixels, 0, width, 0, 0, width, height);

            /** 5.为二维码添加logo图标 */
            if (logoBitmap != null) {
                return addLogo(bitmap, logoBitmap, logoPercent, logoBackgroundColor);
            }
            return bitmap;
        } catch (WriterException e) {
            e.printStackTrace();
            return null;
        }
    }

    /**
     * 向二维码中间添加logo图片(图片合成)
     *
     * @param srcBitmap 原图片（生成的简单二维码图片）
     * @param logoBitmap logo图片
     * @param logoPercent 百分比 (用于调整logo图片在原图片中的显示大小, 取值范围[0,1] )
     * 原图片是二维码时,建议使用0.2F,百分比过大可能导致二维码扫描失败。
     * @param logoBackgroundColor
     * @return
     */
    @Nullable
    public static Bitmap addLogo(@Nullable Bitmap srcBitmap, @Nullable Bitmap logoBitmap,
        float logoPercent, int logoBackgroundColor) {
        if (srcBitmap == null) {
            return null;
        }
        if (logoBitmap == null) {
            return srcBitmap;
        }

        // 二维码高度
        int srcWidth = srcBitmap.getWidth();
        int srcHeight = srcBitmap.getHeight();

        // 画二维码
        Bitmap bitmap = Bitmap.createBitmap(srcWidth, srcHeight, Bitmap.Config.ARGB_8888);
        Canvas canvas = new Canvas(bitmap);
        canvas.drawBitmap(srcBitmap, 0, 0, null);

        // 计算logo高度
        int logoWidth = logoBitmap.getWidth();
        int logoHeight = logoBitmap.getHeight();

        //传值不合法时使用0.2F
        if (logoPercent < 0F || logoPercent > 1F) {
            logoPercent = 0.2F;
        }

        // 画正方形背景
        if (logoBackgroundColor != 0) {
            float logoSize = srcWidth * logoPercent;
            float halfSize = logoSize / 2;
            Paint paint = new Paint();
            paint.setColor(logoBackgroundColor);
            canvas.drawRect(srcWidth / 2f - halfSize, srcHeight / 2f - halfSize,
                srcWidth / 2f + halfSize, srcHeight / 2f + halfSize, paint);
        }

        // 画logo
        float scaleWidth = srcWidth * logoPercent / logoWidth;
        canvas.scale(scaleWidth, scaleWidth, srcWidth / 2f, srcHeight / 2f);

        canvas.drawBitmap(logoBitmap, srcWidth / 2f - logoWidth / 2f,
            srcHeight / 2f - logoHeight / 2f, null);

        return bitmap;
    }
}
